package com.zheng.config;

import java.io.FileInputStream;
import java.lang.reflect.Method;

/**
 * @author: ztl
 * @date: 2023/11/18 22:29
 * @desc: 自己的自定义类加载器
 *  自定义类加载器只需要继承 java.lang.ClassLoader 类，这个类由俩个核心方法：
 *      loadClass(String,boolean)   实现了双亲委派机制
 *      findClass       实现了空方法，(所以我们自定义类加载器主要是重写findClass方法)
 */
public class MyClassLoaderTest {

    static public class MyClassLoader extends ClassLoader {

        private String classPath;

        public MyClassLoader(String classPath){
            this.classPath = classPath;
        }

        /**
         * 这个是读取class包用的
         * @param name
         * @return
         * @throws Exception
         */
        private byte[] loadByte(String name) throws Exception {
            name = name.replaceAll("\\.", "/");
            FileInputStream fis = new FileInputStream(classPath + "/" + name+ ".class");
            int len = fis.available();
            byte[] data = new byte[len];
            fis.read(data);
            fis.close();
            return data;
        }

        /**
         *
         * @param name
         * @return 这个是自定义加载器要用的
         * @date: 2023/11/19 17:38
         * content:
         *  name就是类的全限定路径
         *  defineClass 是ClassLoader中的方法，
         */
        protected Class<?> findClass(String name) throws ClassNotFoundException {
            try {
                byte[] data = loadByte(name);
                //name 类名。 data 类的二进制数组文件
                return defineClass(name, data, 0, data.length);
            } catch (Exception e) {
                e.printStackTrace();
                throw new ClassNotFoundException();
            }
        }

        /**
         * 这个是打破双亲委派机制用的
         * 正常的loadClass是双亲委派式的读取class文件。
         * 但是你现在不是想打破双亲委派嘛，那就重写java的双亲委派，走自己的加载程序，就可以打破了
         *
         */
        protected Class<?> loadClass(String name, boolean resolve)
                throws ClassNotFoundException
        {
            synchronized (getClassLoadingLock(name)) {
                // First, check if the class has already been loaded
                Class<?> c = findLoadedClass(name);
                if (c == null) {
                    long t0 = System.nanoTime();
                    // 这几行是双亲委派的代码,注掉他，直接让他 c==null，走自己的findClass
//                    try {
//                        if (parent != null) {
//                            c = parent.loadClass(name, false);
//                        } else {
//                            c = findBootstrapClassOrNull(name);
//                        }
//                    } catch (ClassNotFoundException e) {
//                        // ClassNotFoundException thrown if class not found
//                        // from the non-null parent class loader
//                    }

                    // 但是由于User1，继承了Object类，所以初始化user的时候，就会初始Object，
                    // 由于Object是由BooStrapClassLoader加载的，你没法直接不走双亲委派，所以呢，加个判断，
                    // 如果是我们自己的类，则走自己的类加载器去加载，如果是Object之类的类，就走原来的双亲委派机制
                    if (!name.startsWith("com.zheng.po")){
                        // 走jdk的双亲委派机制
                        c = this.getParent().loadClass(name);
                    }else {
                        // 走自己重写的findClass类
                        c = findClass(name);
                    }

                    if (c == null) {
                        // If still not found, then invoke findClass in order
                        // to find the class.
                        long t1 = System.nanoTime();
                        c = findClass(name);

                        // this is the defining class loader; record the stats
                        sun.misc.PerfCounter.getParentDelegationTime().addTime(t1 - t0);
                        sun.misc.PerfCounter.getFindClassTime().addElapsedTimeFrom(t1);
                        sun.misc.PerfCounter.getFindClasses().increment();
                    }
                }
                if (resolve) {
                    resolveClass(c);
                }
                return c;
            }
        }

    }

    /**
     *
     * @content 测试，自定义类加载器
     * @return
     * @date: 2023/11/19 18:07
     */
    public static void main(String args[]) throws Exception {
        //初始化自定义类加载器，会先初始化父类ClassLoader，其中会把自定义类加载器的父加载器设置为应用程序类加载器AppClassLoader
        // 自定义类加载器的，加载路径。
        MyClassLoader classLoader = new MyClassLoader("D:/test");
        //D盘创建 com/zheng/po 几级目录，将User类的复制类User1.class丢入该目录
        // 1、这个 com/zheng/po 就是你user1的包路径， 2、这个必须是.class文件，不能是 .java文件
        Class clazz = classLoader.loadClass("com.zheng.po.User1");
        Object obj = clazz.newInstance();
        Method method = clazz.getDeclaredMethod("sout", null);
        method.invoke(obj, null);

        // 父加载器有，就用父加载器的，然后直接返回；父加载器没有，就用子加载器的结果
        // 如果你的appClassLoader就有class1的话，就会打印出来 sun.misc.Launcher$AppClassLoader
        // 如果你的AppClassLoader里面没有的话，就会走自己的 MyClassLoader。com.zheng.config.MyClassLoaderTest$MyClassLoader
        // 因为你的项目内现在有一个 com.zheng.po.user1.class
        // 然后d盘下面还有一个 com.zheng.po.user1.class
        // 所以，他们的类名一模一样，找到一个，就直接返回了。就跟你自己手写一个String类，和java的String类一样，用上级的类加载器的String类

        // 如果已经重写了 loadClass ，那么就是打破双亲委派机制的 com.zheng.config.MyClassLoaderTest$MyClassLoader
        System.out.println(clazz.getClassLoader().getClass().getName());


        System.out.println("=================================================");

        // tomcat就是这么实现的：用同一个类(MyClassLoader)的两个实现(classLoader1、classLoader2)，去加载不同的类
        MyClassLoader classLoader1 = new MyClassLoader("D:/test1");
        Class clazz1 = classLoader1.loadClass("com.zheng.po.User2");
        Object obj1 = clazz1.newInstance();
        Method method1 = clazz1.getDeclaredMethod("sout", null);
        method1.invoke(obj1, null);
        System.out.println(clazz1.getClassLoader().getClass().getName());
    }


}
